graph in python